#!/bin/bash

# 1. 检查并克隆仓库
if [ ! -d "llm-benchmark" ]; then
    echo "正在克隆 llm-benchmark 仓库..."
    git clone https://github.com/lework/llm-benchmark.git
else
    echo "llm-benchmark 文件夹已存在，跳过克隆"
fi

cd llm-benchmark || { echo "无法进入 llm-benchmark 目录"; exit 1; }

if [ ! -d "vllm_venv" ]; then
    echo "正在创建虚拟环境..."
    python3 -m venv vllm_venv
else
    echo "虚拟环境已存在，跳过创建"
fi

# 激活虚拟环境
echo "激活虚拟环境..."
source vllm_venv/bin/activate

# 4. 检查并安装依赖包
echo "检查并安装依赖包..."
pip install openai numpy rich

# 运行 benchmark
echo "开始运行 benchmark..."
python3 ~/llm-benchmark/run_benchmarks.py \
    --llm_url "http://localhost:5536/v1" \
    --api_key "Amdyes@wuhan027" \
    --model "DeepSeek-R1" \
    --use_long_context